// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/spdy/spdy_pinnable_buffer_piece.h"

#include "net/spdy/spdy_prefixed_buffer_reader.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {

namespace test {

    using base::StringPiece;

    class SpdyPinnableBufferPieceTest : public ::testing::Test {
    protected:
        SpdyPrefixedBufferReader Build(const std::string& prefix,
            const std::string& suffix)
        {
            prefix_ = prefix;
            suffix_ = suffix;
            return SpdyPrefixedBufferReader(prefix_.data(), prefix_.length(),
                suffix_.data(), suffix_.length());
        }
        std::string prefix_, suffix_;
    };

    TEST_F(SpdyPinnableBufferPieceTest, Pin)
    {
        SpdyPrefixedBufferReader reader = Build("foobar", "");
        SpdyPinnableBufferPiece piece;
        EXPECT_TRUE(reader.ReadN(6, &piece));

        // Piece points to underlying prefix storage.
        EXPECT_EQ(StringPiece("foobar"), piece);
        EXPECT_FALSE(piece.IsPinned());
        EXPECT_EQ(prefix_.data(), piece.buffer());

        piece.Pin();

        // Piece now points to allocated storage.
        EXPECT_EQ(StringPiece("foobar"), piece);
        EXPECT_TRUE(piece.IsPinned());
        EXPECT_NE(prefix_.data(), piece.buffer());

        // Pinning again has no effect.
        const char* buffer = piece.buffer();
        piece.Pin();
        EXPECT_EQ(buffer, piece.buffer());
    }

    TEST_F(SpdyPinnableBufferPieceTest, Swap)
    {
        SpdyPrefixedBufferReader reader = Build("foobar", "");
        SpdyPinnableBufferPiece piece1, piece2;
        EXPECT_TRUE(reader.ReadN(4, &piece1));
        EXPECT_TRUE(reader.ReadN(2, &piece2));

        piece1.Pin();

        EXPECT_EQ(StringPiece("foob"), piece1);
        EXPECT_TRUE(piece1.IsPinned());
        EXPECT_EQ(StringPiece("ar"), piece2);
        EXPECT_FALSE(piece2.IsPinned());

        piece1.Swap(&piece2);

        EXPECT_EQ(StringPiece("ar"), piece1);
        EXPECT_FALSE(piece1.IsPinned());
        EXPECT_EQ(StringPiece("foob"), piece2);
        EXPECT_TRUE(piece2.IsPinned());

        SpdyPinnableBufferPiece empty;
        piece2.Swap(&empty);

        EXPECT_EQ(StringPiece(""), piece2);
        EXPECT_FALSE(piece2.IsPinned());
    }

} // namespace test

} // namespace net
